# vim:set ft= ts=4 sw=4 et:

use Test::Nginx::Socket::Lua;
use Cwd qw(cwd);

repeat_each(2);

plan tests => repeat_each() * (3 * blocks());

my $pwd = cwd();

our $HttpConfig = qq{
    lua_package_path "$pwd/lib/?.lua;;";
    lua_package_cpath "/usr/local/openresty-debug/lualib/?.so;/usr/local/openresty/lualib/?.so;;";
};

$ENV{TEST_NGINX_RESOLVER} = '8.8.8.8';
$ENV{TEST_NGINX_SSDB_PORT} ||= 8888;

no_long_string();
#no_diff();

run_tests();

__DATA__

=== TEST 1: set and get
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end


            local res, err = db:set("dog", "an animal")
            if not res then
                ngx.say("failed to set dog: ", err)
                return
            end

            ngx.say("set dog: ", res)

            for i = 1, 2 do
                local res, err = db:get("dog")
                if err then
                    ngx.say("failed to get dog: ", err)
                    return
                end

                if not res then
                    ngx.say("dog not found.")
                    return
                end

                ngx.say("dog: ", res)
            end

            db:close()
        ';
    }
--- request
GET /t
--- response_body
set dog: true
dog: an animal
dog: an animal
--- no_error_log
[error]


=== TEST 2: flushdb
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if err ~= nil then
                ngx.say("failed to flushdb: ", err)
                return
            end
            ngx.say("flushdb: ", true)

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb: true
--- no_error_log
[error]



=== TEST 3: get nil bulk value
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if err ~= nil  then
                ngx.say("failed to flushdb: ", err)
                return
            end

            ngx.say("flushdb success")

            for i = 1, 2 do
                res, err = db:get("not_found")
                if err then
                    ngx.say("failed to get: ", err)
                    return
                end

                if res == ngx.null then
                    ngx.say("not_found not found.")
                    return
                end

                ngx.say("get not_found: ", res)
            end

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb success
not_found not found.
--- no_error_log
[error]



=== TEST 4: get nil list
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if err ~= nil then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if err ~= nil then
                ngx.say("failed to flushdb: ", err)
                return
            end

            ngx.say("flushdb success")

            for i = 1, 2 do
                res, err = db:qslice("nokey", 0, 1)
                if err then
                    ngx.say("failed to get: ", err)
                    return
                end

                if res == ngx.null then
                    ngx.say("nokey not found.")
                    return
                end

                ngx.say("get nokey: ", #res, " (", type(res), ")")
            end

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb success
get nokey: 0 (table)
get nokey: 0 (table)
--- no_error_log
[error]



=== TEST 5: incr and decr
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:set("connections", 10)
            if not res then
                ngx.say("failed to set connections: ", err)
                return
            end

            ngx.say("set connections: ", res)

            res, err = db:incr("connections")
            if not res then
                ngx.say("failed to set connections: ", err)
                return
            end

            ngx.say("incr connections: ", res)

            local res, err = db:get("connections")
            if err then
                ngx.say("failed to get connections: ", err)
                return
            end

            res, err = db:incr("connections")
            if not res then
                ngx.say("failed to incr connections: ", err)
                return
            end

            ngx.say("incr connections: ", res)

            res, err = db:decr("connections")
            if not res then
                ngx.say("failed to decr connections: ", err)
                return
            end

            ngx.say("decr connections: ", res)

            res, err = db:get("connections")
            if not res then
                ngx.say("connections not found.")
                return
            end

            ngx.say("connections: ", res)

            res, err = db:del("connections")
            if not res then
                ngx.say("failed to del connections: ", err)
                return
            end

            ngx.say("del connections: ", res)

            res, err = db:incr("connections")
            if not res then
                ngx.say("failed to set connections: ", err)
                return
            end

            ngx.say("incr connections: ", res)

            res, err = db:get("connections")
            if not res then
                ngx.say("connections not found.")
                return
            end

            ngx.say("connections: ", res)

            db:close()
        ';
    }
--- request
GET /t
--- response_body
set connections: true
incr connections: 11
incr connections: 12
decr connections: 11
connections: 11
del connections: true
incr connections: 1
connections: 1
--- no_error_log
[error]



=== TEST 6: bad incr command format
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:incr("connections", 12)
            if not res then
                ngx.say("failed to set connections: ", res, ": ", err)
                return
            end

            ngx.say("incr connections: ", res)

            db:close()
        ';
    }
--- request
GET /t
--- response_body
failed to set connections: false: ERR wrong number of arguments for 'incr' command
--- no_error_log
[error]
--- SKIP



=== TEST 7: qpush_front and qslice
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if err ~= nil then
                ngx.say("failed to flushdb: ", err)
                return
            end
            ngx.say("flushdb: OK")

            local res, err = db:qpush_front("mylist", "world")
            if not res then
                ngx.say("failed to lpush: ", err)
                return
            end
            ngx.say("qpush_front result: ", res)

            res, err = db:qpush_front("mylist", "hello")
            if not res then
                ngx.say("failed to qpush_front: ", err)
                return
            end
            ngx.say("qpush_front result: ", res)

            res, err = db:qslice("mylist", 0, -1)
            if not res then
                ngx.say("failed to qslice: ", err)
                return
            end
            local cjson = require "cjson"
            ngx.say("qslice result: ", cjson.encode(res))

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb: OK
qpush_front result: 1
qpush_front result: 2
qslice result: ["hello","world"]
--- no_error_log
[error]



=== TEST 8: blpop expires its own timeout (ssdb doesn't support blpop)
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(2500) -- 2.5 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if err ~= nil then
                ngx.say("failed to flushdb: ", err)
                return
            end
            ngx.say("flushdb: OK")

            local res, err = db:blpop("key", 1)
            if err then
                ngx.say("failed to blpop: ", err)
                return
            end

            if res == ngx.null then
                ngx.say("no element popped.")
                return
            end

            local cjson = require "cjson"
            ngx.say("blpop result: ", cjson.encode(res))

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb: OK
no element popped.
--- no_error_log
[error]
--- timeout: 3
--- SKIP


=== TEST 9: blpop expires cosocket timeout
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:flushdb()
            if not res then
                ngx.say("failed to flushdb: ", err)
                return
            end
            ngx.say("flushdb: ", res)

            db:set_timeout(200) -- 200 ms

            local res, err = db:blpop("key", 1)
            if err then
                ngx.say("failed to blpop: ", err)
                return
            end

            if not res then
                ngx.say("no element popped.")
                return
            end

            local cjson = require "cjson"
            ngx.say("blpop result: ", cjson.encode(res))

            db:close()
        ';
    }
--- request
GET /t
--- response_body
flushdb: OK
failed to blpop: timeout
--- error_log
lua tcp socket read timed out
--- SKIP



=== TEST 10: set keepalive and get reused times
--- http_config eval: $::HttpConfig
--- config
    resolver $TEST_NGINX_RESOLVER;
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local times = db:get_reused_times()
            ngx.say("reused times: ", times)

            local ok, err = db:set_keepalive()
            if not ok then
                ngx.say("failed to set keepalive: ", err)
                return
            end

            ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            times = db:get_reused_times()
            ngx.say("reused times: ", times)
        ';
    }
--- request
GET /t
--- response_body
reused times: 0
reused times: 1
--- no_error_log
[error]



=== TEST 11: multi_get
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ok, err = db:flushdb()
            if err ~= nil then
                ngx.say("failed to flush all: ", err)
                return
            end

            local res, err = db:set("dog", "an animal")
            if not res then
                ngx.say("failed to set dog: ", err)
                return
            end

            ngx.say("set dog: ", res)

            for i = 1, 2 do
                local res, err = db:multi_get("dog", "cat", "dog")
                if err then
                    ngx.say("failed to get dog: ", err)
                    return
                end

                if not res then
                    ngx.say("dog not found.")
                    return
                end

                local cjson = require "cjson"
                ngx.say("res: ", cjson.encode(res))
            end

            db:close()
        ';
    }
--- request
GET /t
--- response_body
set dog: true
res: {"dog":"an animal"}
res: {"dog":"an animal"}
--- no_error_log
[error]




=== TEST 12: hmget array_to_hash
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()
            local cjson = require "cjson"

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ok, err = db:flushdb()
            if not ok then
                ngx.say("failed to flush all: ", err)
                return
            end

            local res, err = db:multi_hset("animals", { dog = "bark", cat = "meow", cow = "moo" })
            if not res then
                ngx.say("failed to set animals: ", err)
                return
            end

            ngx.say("hmset animals: ", cjson.encode(res))

            local res, err = db:multi_hget("animals", "dog", "cat", "cow")
            if not res then
                ngx.say("failed to get animals: ", err)
                return
            end

            ngx.say("hmget animals: ", cjson.encode(res))

            local res, err = db:hgetall("animals")
            if err then
                ngx.say("failed to get animals: ", err)
                return
            end

            if not res then
                ngx.say("animals not found.")
                return
            end

            ngx.say("dog: ", res.dog)
            ngx.say("cat: ", res.cat)
            ngx.say("cow: ", res.cow)

            db:close()
        ';
    }
--- request
GET /t
--- response_body
hmset animals: 3
hmget animals: {"dog":"bark","cat":"meow","cow":"moo"}
dog: bark
cat: meow
cow: moo
--- no_error_log
[error]



=== TEST 13: boolean args
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ok, err = db:set("foo", true)
            if not ok then
                ngx.say("failed to set: ", err)
                return
            end

            local res, err = db:get("foo")
            if not res then
                ngx.say("failed to get: ", err)
                return
            end

            ngx.say("foo: ", res, ", type: ", type(res))

            ok, err = db:set("foo", false)
            if not ok then
                ngx.say("failed to set: ", err)
                return
            end

            local res, err = db:get("foo")
            if not res then
                ngx.say("failed to get: ", err)
                return
            end

            ngx.say("foo: ", res, ", type: ", type(res))

            ok, err = db:set("foo", nil)
            if not ok then
                ngx.say("failed to set: ", err)
            end

            local res, err = db:get("foo")
            if not res then
                ngx.say("failed to get: ", err)
                return
            end

            ngx.say("foo: ", res, ", type: ", type(res))

            local ok, err = db:set_keepalive(10, 10)
            if not ok then
                ngx.say("failed to set_keepalive: ", err)
            end
        ';
    }
--- request
GET /t
--- response_body
foo: true, type: string
foo: false, type: string
failed to set: wrong number of arguments
foo: false, type: string
--- no_error_log
[error]



=== TEST 14: set and get (key with underscores)
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            local res, err = db:set("a_dog", "an animal")
            if not res then
                ngx.say("failed to set a_dog: ", err)
                return
            end

            ngx.say("set a_dog: ", res)

            for i = 1, 2 do
                local res, err = db:get("a_dog")
                if err then
                    ngx.say("failed to get a_dog: ", err)
                    return
                end

                if not res then
                    ngx.say("a_dog not found.")
                    return
                end

                ngx.say("a_dog: ", res)
            end

            db:close()
        ';
    }
--- request
GET /t
--- response_body
set a_dog: true
a_dog: an animal
a_dog: an animal
--- no_error_log
[error]



=== TEST 15: connection refused
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()

            db:set_timeout(10000) -- 10 sec

            local ok, err = db:connect("127.0.0.1", 81)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ngx.say("connected")

            db:close()
        ';
    }
--- request
GET /t
--- response_body
failed to connect: connection refused
--- timeout: 3
--- no_error_log
[alert]



=== TEST 16: hexists
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()
            local cjson = require "cjson"

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ok, err = db:flushdb()
            if not ok then
                ngx.say("failed to flush all: ", err)
                return
            end

            local res, err = db:hexists("animals", "dog")
            ngx.say("hexists animals dog: ", res)

            local res, err = db:multi_hset("animals", { dog = "bark", cat = "meow", cow = "moo" })
            if not res then
                ngx.say("failed to set animals: ", err)
                return
            end
            local res, err = db:hexists("animals", "dog")
            ngx.say("hexists animals dog: ", cjson.encode(res))

            db:close()
        ';
    }
--- request
GET /t
--- response_body
hexists animals dog: false
hexists animals dog: true
--- no_error_log
[error]



=== TEST 16: pipeline
--- http_config eval: $::HttpConfig
--- config
    location /t {
        content_by_lua '
            local ssdb = require "resty.ssdb"
            local db = ssdb:new()
            local cjson = require "cjson"

            db:set_timeout(1000) -- 1 sec

            local ok, err = db:connect("127.0.0.1", $TEST_NGINX_SSDB_PORT)
            if not ok then
                ngx.say("failed to connect: ", err)
                return
            end

            ok, err = db:flushdb()
            if not ok then
                ngx.say("failed to flush all: ", err)
                return
            end


            db:init_pipeline()
            db:set("cat", "Marry")
            db:set("horse", "Bob")
            db:get("cat")
            db:get("horse")
            db:hset("hash1", "k1", "v1")
            db:hset("hash1", "k2", "v2")
            db:hget("hash1", "k1")
            db:hgetall("hash1")

            local res, err = db:commit_pipeline()
            if not res then
               ngx.say("failed to commit the pipelined requests: ", err)
               return
            end
            ngx.say("pipeline response: ", cjson.encode(res))

            db:close()
        ';
    }
--- request
GET /t
--- response_body
pipeline response: [true,true,"Marry","Bob",true,true,"v1",{"k1":"v1","k2":"v2"}]
--- no_error_log
[error]
